package com.neusoft.news.controller;

import java.io.File;
import java.io.IOException;
import java.util.UUID;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import javax.servlet.http.Part;

import org.apache.commons.fileupload.servlet.ServletFileUpload;

import com.neusoft.news.bean.News;
import com.neusoft.news.service.NewsService;

import java.io.PrintWriter;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;

/**
 * Servlet implementation class AddNewsServlet
 */
@WebServlet("/AddNewsServlet")
public class AddNewsServlet extends HttpServlet {
	private static final long serialVersionUID = 1L;
	// 上传文件存储目录
    private static final String UPLOAD_DIRECTORY = "img";
 
    // 上传配置
    private static final int MEMORY_THRESHOLD   = 1024 * 1024 * 3;  // 3MB
    private static final int MAX_FILE_SIZE      = 1024 * 1024 * 40; // 40MB
    private static final int MAX_REQUEST_SIZE   = 1024 * 1024 * 50; // 50MB
       
    /**
     * @see HttpServlet#HttpServlet()
     */
    public AddNewsServlet() {
        super();
        // TODO Auto-generated constructor stub
    }

	/**
	 * @see HttpServlet#doGet(HttpServletRequest request, HttpServletResponse response)
	 */
	protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		// TODO Auto-generated method stub
		HttpSession session = request.getSession();
		response.setContentType("text/html;charset=utf-8");
		request.setCharacterEncoding("UTF-8");
		response.setCharacterEncoding("UTF-8");
		//定义map存储表单中非文件数据
		Map<String, String> map= new HashMap<String, String>();
		//定义news存储最终add的新闻数据
		News news = new News();
//		// 检测是否为多媒体上传
//        if (!ServletFileUpload.isMultipartContent(request)) {
//            // 如果不是则停止
//            PrintWriter writer = response.getWriter();
//            writer.println("Error: 表单必须包含 enctype=multipart/form-data");
//            writer.flush();
//            return;
//        }
     // 配置上传参数
        DiskFileItemFactory factory = new DiskFileItemFactory();
        // 设置内存临界值 - 超过后将产生临时文件并存储于临时目录中
        factory.setSizeThreshold(MEMORY_THRESHOLD);
        // 设置临时存储目录
        factory.setRepository(new File(System.getProperty("java.io.tmpdir")));
 
        ServletFileUpload upload = new ServletFileUpload(factory);
         
        // 设置最大文件上传值
        upload.setFileSizeMax(MAX_FILE_SIZE);
         
        // 设置最大请求值 (包含文件和表单数据)
        upload.setSizeMax(MAX_REQUEST_SIZE);

        // 中文处理
        upload.setHeaderEncoding("UTF-8"); 
     // 构造临时路径来存储上传的文件
        // 这个路径相对当前应用的目录
//        String uploadPath = request.getSession().getServletContext().getRealPath("/") + UPLOAD_DIRECTORY;
        String uploadPath = "D:\\workspace3\\newsweb\\src\\main\\webapp\\img";
//        String uploadPath = "/WEB-INF/jsp" + UPLOAD_DIRECTORY;
       
        // 如果目录不存在则创建
        File uploadDir = new File(uploadPath);
        if (!uploadDir.exists()) {
            uploadDir.mkdir();
        }
        try {
            // 解析请求的内容提取文件数据
            @SuppressWarnings("unchecked")
            List<FileItem> formItems = upload.parseRequest(request);
 
            if (formItems != null && formItems.size() > 0) {
                // 迭代表单数据
                for (FileItem item : formItems) {
                    // 处理不在表单中的字段
                    if (!item.isFormField()) {
                        String fileName = UUID.randomUUID()+new File(item.getName()).getName();
                        String filePath = uploadPath + File.separator + fileName;
                        File storeFile = new File(filePath);
                        // 在控制台输出文件的上传路径
                        System.out.println(filePath);
                        // 保存文件到硬盘
                        item.write(storeFile);
                        //将文件名存入news_photo属性中
                		news.setNews_photo(fileName);
                        request.setAttribute("message","文件上传成功!");
                    } else {
                       /**
                                            * 获取表单中的非文件值
                         */
                    	//表单中的空间name值
                        System.out.println(item.getFieldName());
                        //该name值空间中的value值
                        System.out.println(item.getString("UTF-8"));
                        map.put(item.getFieldName(), item.getString("UTF-8"));
                    } 
                }
            }
        } catch (Exception ex) {
            request.setAttribute("message",
                    "错误信息: " + ex.getMessage());
        }
		
		
		//若未填写id则自动生成id
		if(map.get("news_id")!=null && !map.get("news_id").equals("")) {
			news.setNews_id(map.get("news_id"));
		} else {
			news.setNews_id(UUID.randomUUID()+"");
		}
		news.setNews_title(map.get("news_title"));
		news.setNews_intro(map.get("news_intro"));
		news.setNews_content(map.get("news_content"));
		news.setColumns_id(map.get("columns_id"));
		news.setNews_place(map.get("news_place"));
		news.setNews_important(map.get("news_important"));
		//新增的新闻均为未审核状态
		news.setNews_state("0");
		news.setNews_author(map.get("news_author"));
		NewsService newsService = new NewsService();
		//num：数据库中新增数据个数，若成功添加则num为1，否则为0
		int num = newsService.insertNews(news);
		System.out.println("Servlet add:"+news);
		if(num>0) {
			request.setAttribute("news_id", news.getNews_id());
//			request.getRequestDispatcher("ShowNewsListServlet").forward(request, response);
			response.sendRedirect("ShowNewsListServlet");
		}else {
			String msg = "添加失败";
			request.setAttribute("msg", msg);
			request.getRequestDispatcher("GoToAddNewsPageServlet").forward(request, response);
		}
	}

	/**
	 * @see HttpServlet#doPost(HttpServletRequest request, HttpServletResponse response)
	 */
	protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		// TODO Auto-generated method stub
		doGet(request, response);
	}

}
